package org.example

import org.apache.spark.SparkConf
import org.apache.spark.streaming.dstream.{DStream, ReceiverInputDStream}
import org.apache.spark.streaming.{Seconds, StreamingContext}
object aaaaa {
  def main(args: Array[String]): Unit = {
     val sparkConf = new SparkConf().setMaster("local[2]").setAppName("NetworkWordCount") // 5秒统计一次
     val ssc = new StreamingContext(sparkConf, Seconds(5)) //数据处理
     ssc.checkpoint("checkpoint_dir")
     val dStream: ReceiverInputDStream[String] = ssc.socketTextStream("172.16.104.54",9999) //  词频统计
     val lines: DStream[String] = dStream.flatMap(_.split(" "))
//     val wordDStream: DStream[(String,Int)] = lines.map((_,1))
//     val result = wordDStream.reduceByKey(_+_) //数据保存
     // 定义累加函数
      val wordCounts = lines.map(word => (word, 5))
     val updateFunc = (values: Seq[Int], state: Option[Int]) => {
        val currentCount = values.sum
        val previousCount = state.getOrElse(0)
        Some(currentCount + previousCount)
      }
     val runningCounts = wordCounts.updateStateByKey(updateFunc)
     runningCounts.print()
     ssc.start()
     ssc.awaitTermination()
    }
}
